import torch
import numpy as np
import visdom

a = np.zeros([2, 2])

out = torch.from_numpy(a)
print(out)
print(out.shape)
vis = visdom.Visdom()
vis.text("Hello")
# vis.image(a)
vis.image(np.ones((3, 10, 10)))